题目要求
Suppose you have a random list of people standing in a queue. Each person is described by a pair of integers (h, k), where h is the height of the person and k is the number of people in front of this person who have a height greater than or equal to h. Write an algorithm to reconstruct the queue.
Note:
The number of people is less than 1,100.
Example
Input:
[[7,0], [4,4], [7,1], [5,0], [6,1], [5,2]]
Output:
[[5,0], [7,0], [5,2], [6,1], [4,4], [7,1]]
假设有一组人站成一堆,每个人都记录下了自己的高度,以及在自己前面有多少个不比自己矮的人。现在请按照这个信息将这组人放在队列中正确的位置上并返回。
思路和代码
刚开始我试图用分治法来解决,因为每一个小队列中,高度最矮且站在自己前面的高个最少的人一定位于k位置上。但是这样解决其实复杂化了问题。
可以从另一个角度来想,首先我们知道,同等高度的人,其相对位置一定是根据k的值从小到大排列的。即k越大,则该同等高度的人一定在另一个同等高度的人后面。如果我们从高到低将人们插入队列中,可以知道,k的值就该人在当前队列中的下标,如:
[[7,0], [4,4], [7,1], [5,0], [6,1], [5,2]]
首先将其按照h和k排序,得出结果:
[[7,0],[7,1],[6,1],[5,1],[5,0],[5,2],[4,4]]
当前结果队列为[]
将[7,0]插入下标为0的位置上 结果队列[[7,0]]
将[7,1]插入下标为1的位置上 结果队列[[7,0],[7,1]]
将[6,1]插入下标为1的位置上 结果队列[[7,0],[6,1],[7,1]]
按照这种规则,依次按照顺序和k的值将数据插入结果队列中
代码如下:
public int[][] reconstructQueue(int[][] people) {
int[][] result = new int[people.length][2];
Arrays.sort(people, new Comparator<int[]>() {
@Override
public int compare(int[] o1, int[] o2) {
return o1[0] == o2[0] ? o1[1] - o2[1] : o2[0] - o1[0];
}
});
for(int i = 0 ; i<people.length ; i++) {
int pos = people[i][1];
for (int j = i; j > pos; j--) {
result[j] = result[j - 1];
}
result[pos] = people[i];
}
return result;
}
**粗体** _斜体_ [链接](http://example.com) `代码` - 列表 > 引用
。你还可以使用@
来通知其他用户。